Node / ReticulumProjects / MeshChatX.git / files / tests / backend / test_database.py

Displaying Raw • Download

Tff7b72import T7ee787os

Tff7b72import T7ee787sqlite3

Tff7b72import T7ee787tempfile

Tff7b72import T7ee787pytest

Tff7b72from T7ee787meshchatxT7ee787.T7ee787srcT7ee787.T7ee787backendT7ee787.T7ee787databaseT7ee787.T7ee787legacy_migrator Tff7b72import Te6edf3LegacyMigrator

Tff7b72from T7ee787meshchatxT7ee787.T7ee787srcT7ee787.T7ee787backendT7ee787.T7ee787databaseT7ee787.T7ee787provider Tff7b72import Te6edf3DatabaseProvider

Tff7b72from T7ee787meshchatxT7ee787.T7ee787srcT7ee787.T7ee787backendT7ee787.T7ee787databaseT7ee787.T7ee787schema Tff7b72import Te6edf3DatabaseSchema

Tf0883e@pytestTff7b72.Td2a8fffixture

Tff7b72def Td2a8fftemp_dbTb4b4b4(Tb4b4b4)Tb4b4b4:

Te6edf3fdTb4b4b4, Te6edf3path Tff7b72= Te6edf3tempfileTff7b72.Td2a8ffmkstempTb4b4b4(Tb4b4b4)

Te6edf3osTff7b72.Td2a8ffcloseTb4b4b4(Te6edf3fdTb4b4b4)

Tff7b72yield Te6edf3path

Tff7b72if Te6edf3osTff7b72.Td2a8ffpathTff7b72.Td2a8ffexistsTb4b4b4(Te6edf3pathTb4b4b4)Tb4b4b4:

Te6edf3osTff7b72.Td2a8ffremoveTb4b4b4(Te6edf3pathTb4b4b4)

Tff7b72def Td2a8fftest_database_initializationTb4b4b4(Te6edf3temp_dbTb4b4b4)Tb4b4b4:

Te6edf3provider Tff7b72= Te6edf3DatabaseProviderTb4b4b4(Te6edf3temp_dbTb4b4b4)

Te6edf3schema Tff7b72= Te6edf3DatabaseSchemaTb4b4b4(Te6edf3providerTb4b4b4)

Te6edf3schemaTff7b72.Td2a8ffinitializeTb4b4b4(Tb4b4b4)

T8b949e# Check if tables were created

Te6edf3tables Tff7b72= Te6edf3providerTff7b72.Td2a8fffetchallTb4b4b4(Ta5d6ff"Ta5d6ffSELECT name FROM sqlite_master WHERE type=Ta5d6ff'Ta5d6fftableTa5d6ff'Ta5d6ff"Tb4b4b4)

Te6edf3table_names Tff7b72= Tb4b4b4[Te6edf3rowTb4b4b4[Ta5d6ff"Ta5d6ffnameTa5d6ff"Tb4b4b4] Tff7b72for Te6edf3row Tff7b72in Te6edf3tablesTb4b4b4]

Tff7b72assert Ta5d6ff"Ta5d6ffconfigTa5d6ff" Tff7b72in Te6edf3table_names

Tff7b72assert Ta5d6ff"Ta5d6fflxmf_messagesTa5d6ff" Tff7b72in Te6edf3table_names

Tff7b72assert Ta5d6ff"Ta5d6ffannouncesTa5d6ff" Tff7b72in Te6edf3table_names

T8b949e# Check version

Te6edf3version_row Tff7b72= Te6edf3providerTff7b72.Td2a8fffetchoneTb4b4b4(

Ta5d6ff"Ta5d6ffSELECT value FROM config WHERE key = Ta5d6ff'Ta5d6ffdatabase_versionTa5d6ff'Ta5d6ff"Tb4b4b4,

Tb4b4b4)

Tff7b72assert Tffa657intTb4b4b4(Te6edf3version_rowTb4b4b4[Ta5d6ff"Ta5d6ffvalueTa5d6ff"Tb4b4b4]Tb4b4b4) Tff7b72== Te6edf3DatabaseSchemaTff7b72.Td2a8ffLATEST_VERSION

Te6edf3providerTff7b72.Td2a8ffcloseTb4b4b4(Tb4b4b4)

Tff7b72def Td2a8fftest_legacy_migrator_detectionTb4b4b4(Te6edf3temp_dbTb4b4b4)Tb4b4b4:

T8b949e# Setup current DB

Te6edf3provider Tff7b72= Te6edf3DatabaseProviderTb4b4b4(Te6edf3temp_dbTb4b4b4)

Te6edf3schema Tff7b72= Te6edf3DatabaseSchemaTb4b4b4(Te6edf3providerTb4b4b4)

Te6edf3schemaTff7b72.Td2a8ffinitializeTb4b4b4(Tb4b4b4)

T8b949e# Setup a "legacy" DB in a temp directory

Tff7b72with Te6edf3tempfileTff7b72.Td2a8ffTemporaryDirectoryTb4b4b4(Tb4b4b4) Tff7b72as Te6edf3legacy_dirTb4b4b4:

Te6edf3identity_hash Tff7b72= Ta5d6ff"Ta5d6ffdeadbeefTa5d6ff"

Te6edf3legacy_identity_dir Tff7b72= Te6edf3osTff7b72.Td2a8ffpathTff7b72.Td2a8ffjoinTb4b4b4(Te6edf3legacy_dirTb4b4b4, Ta5d6ff"Ta5d6ffidentitiesTa5d6ff"Tb4b4b4, Te6edf3identity_hashTb4b4b4)

Te6edf3osTff7b72.Td2a8ffmakedirsTb4b4b4(Te6edf3legacy_identity_dirTb4b4b4)

Te6edf3legacy_db_path Tff7b72= Te6edf3osTff7b72.Td2a8ffpathTff7b72.Td2a8ffjoinTb4b4b4(Te6edf3legacy_identity_dirTb4b4b4, Ta5d6ff"Ta5d6ffdatabase.dbTa5d6ff"Tb4b4b4)

Te6edf3legacy_conn Tff7b72= Te6edf3sqlite3Tff7b72.Td2a8ffconnectTb4b4b4(Te6edf3legacy_db_pathTb4b4b4)

Te6edf3legacy_connTff7b72.Td2a8ffexecuteTb4b4b4(Ta5d6ff"Ta5d6ffCREATE TABLE config (key TEXT, value TEXT)Ta5d6ff"Tb4b4b4)

Te6edf3legacy_connTff7b72.Td2a8ffexecuteTb4b4b4(

Ta5d6ff"Ta5d6ffINSERT INTO config (key, value) VALUES (Ta5d6ff'Ta5d6ffdisplay_nameTa5d6ff'Ta5d6ff, Ta5d6ff'Ta5d6ffLegacy UserTa5d6ff'Ta5d6ff)Ta5d6ff"Tb4b4b4,

Tb4b4b4)

Te6edf3legacy_connTff7b72.Td2a8ffcommitTb4b4b4(Tb4b4b4)

Te6edf3legacy_connTff7b72.Td2a8ffcloseTb4b4b4(Tb4b4b4)

Te6edf3migrator Tff7b72= Te6edf3LegacyMigratorTb4b4b4(Te6edf3providerTb4b4b4, Te6edf3legacy_dirTb4b4b4, Te6edf3identity_hashTb4b4b4)

Tff7b72assert Te6edf3migratorTff7b72.Td2a8ffget_legacy_db_pathTb4b4b4(Tb4b4b4) Tff7b72== Te6edf3legacy_db_path

Tff7b72assert Te6edf3migratorTff7b72.Td2a8ffshould_migrateTb4b4b4(Tb4b4b4) Tff7b72is Tff7b72True

Te6edf3providerTff7b72.Td2a8ffcloseTb4b4b4(Tb4b4b4)

Tff7b72def Td2a8fftest_legacy_migration_dataTb4b4b4(Te6edf3temp_dbTb4b4b4)Tb4b4b4:

Te6edf3provider Tff7b72= Te6edf3DatabaseProviderTb4b4b4(Te6edf3temp_dbTb4b4b4)

Te6edf3schema Tff7b72= Te6edf3DatabaseSchemaTb4b4b4(Te6edf3providerTb4b4b4)

Te6edf3schemaTff7b72.Td2a8ffinitializeTb4b4b4(Tb4b4b4)

Tff7b72with Te6edf3tempfileTff7b72.Td2a8ffTemporaryDirectoryTb4b4b4(Tb4b4b4) Tff7b72as Te6edf3legacy_dirTb4b4b4:

Te6edf3identity_hash Tff7b72= Ta5d6ff"Ta5d6ffdeadbeefTa5d6ff"

Te6edf3legacy_identity_dir Tff7b72= Te6edf3osTff7b72.Td2a8ffpathTff7b72.Td2a8ffjoinTb4b4b4(Te6edf3legacy_dirTb4b4b4, Ta5d6ff"Ta5d6ffidentitiesTa5d6ff"Tb4b4b4, Te6edf3identity_hashTb4b4b4)

Te6edf3osTff7b72.Td2a8ffmakedirsTb4b4b4(Te6edf3legacy_identity_dirTb4b4b4)

Te6edf3legacy_db_path Tff7b72= Te6edf3osTff7b72.Td2a8ffpathTff7b72.Td2a8ffjoinTb4b4b4(Te6edf3legacy_identity_dirTb4b4b4, Ta5d6ff"Ta5d6ffdatabase.dbTa5d6ff"Tb4b4b4)

T8b949e# Create legacy DB with some data

Te6edf3legacy_conn Tff7b72= Te6edf3sqlite3Tff7b72.Td2a8ffconnectTb4b4b4(Te6edf3legacy_db_pathTb4b4b4)

Te6edf3legacy_connTff7b72.Td2a8ffexecuteTb4b4b4(

Ta5d6ff"Ta5d6ffCREATE TABLE lxmf_messages (hash TEXT UNIQUE, content TEXT)Ta5d6ff"Tb4b4b4,

Tb4b4b4)

Te6edf3legacy_connTff7b72.Td2a8ffexecuteTb4b4b4(

Ta5d6ff"Ta5d6ffINSERT INTO lxmf_messages (hash, content) VALUES (Ta5d6ff'Ta5d6ffmsg1Ta5d6ff'Ta5d6ff, Ta5d6ff'Ta5d6ffHello LegacyTa5d6ff'Ta5d6ff)Ta5d6ff"Tb4b4b4,

Tb4b4b4)

Te6edf3legacy_connTff7b72.Td2a8ffexecuteTb4b4b4(Ta5d6ff"Ta5d6ffCREATE TABLE config (key TEXT UNIQUE, value TEXT)Ta5d6ff"Tb4b4b4)

Te6edf3legacy_connTff7b72.Td2a8ffexecuteTb4b4b4(

Ta5d6ff"Ta5d6ffINSERT INTO config (key, value) VALUES (Ta5d6ff'Ta5d6fftest_keyTa5d6ff'Ta5d6ff, Ta5d6ff'Ta5d6fftest_valTa5d6ff'Ta5d6ff)Ta5d6ff"Tb4b4b4,

Tb4b4b4)

Te6edf3legacy_connTff7b72.Td2a8ffcommitTb4b4b4(Tb4b4b4)

Te6edf3legacy_connTff7b72.Td2a8ffcloseTb4b4b4(Tb4b4b4)

Te6edf3migrator Tff7b72= Te6edf3LegacyMigratorTb4b4b4(Te6edf3providerTb4b4b4, Te6edf3legacy_dirTb4b4b4, Te6edf3identity_hashTb4b4b4)

Tff7b72assert Te6edf3migratorTff7b72.Td2a8ffmigrateTb4b4b4(Tb4b4b4) Tff7b72is Tff7b72True

T8b949e# Verify data moved

Te6edf3msg_row Tff7b72= Te6edf3providerTff7b72.Td2a8fffetchoneTb4b4b4(

Ta5d6ff"Ta5d6ffSELECT content FROM lxmf_messages WHERE hash = Ta5d6ff'Ta5d6ffmsg1Ta5d6ff'Ta5d6ff"Tb4b4b4,

Tb4b4b4)

Tff7b72assert Te6edf3msg_rowTb4b4b4[Ta5d6ff"Ta5d6ffcontentTa5d6ff"Tb4b4b4] Tff7b72== Ta5d6ff"Ta5d6ffHello LegacyTa5d6ff"

Te6edf3config_row Tff7b72= Te6edf3providerTff7b72.Td2a8fffetchoneTb4b4b4(

Ta5d6ff"Ta5d6ffSELECT value FROM config WHERE key = Ta5d6ff'Ta5d6fftest_keyTa5d6ff'Ta5d6ff"Tb4b4b4,

Tb4b4b4)

Tff7b72assert Te6edf3config_rowTb4b4b4[Ta5d6ff"Ta5d6ffvalueTa5d6ff"Tb4b4b4] Tff7b72== Ta5d6ff"Ta5d6fftest_valTa5d6ff"

Te6edf3providerTff7b72.Td2a8ffcloseTb4b4b4(Tb4b4b4)